

package com.gc.server.service.impl;

import com.gc.server.dao.ShopDao;
import com.gc.server.dao.UserDao;
import com.gc.server.entity.ShopEntity;
import com.gc.server.entity.UserEntity;
import com.yf.response.RRException;
import org.springframework.transaction.annotation.Transactional;
import com.gc.server.vo.up.UserShopUp;
import org.springframework.beans.factory.annotation.Autowired;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import org.springframework.stereotype.Service;
import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSON;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.gc.server.vo.down.UserShopDown;

import java.io.InputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.io.IOException;
import java.net.URLEncoder;

import com.gc.server.vo.down.UserShopDown;

import javax.servlet.http.HttpServletResponse;
import java.util.stream.Collectors;

import com.gc.server.dao.UserShopDao;
import com.gc.server.entity.UserShopEntity;
import com.gc.server.vo.UserShopVo;
import com.gc.server.service.UserShopService;
import com.yf.service.impl.BaseServiceImpl;

/**
 * 账户记录Service实现类
 *
 * @author hxf
 * @email 505860922@qq.com
 * @date ${datetime}
 */
@Service
public class UserShopServiceImpl extends BaseServiceImpl<UserShopEntity> implements UserShopService {

    private UserShopDao dao;


    @Autowired
    public void setDao(UserShopDao dao) {
        this.dao = dao;
        super.setBaseDao(dao);
    }


    @Override
    public List<UserShopVo> findVo(UserShopVo vo) {
        return dao.findVo(vo);
    }

    @Override
    public UserShopVo getVo(Long id) {
        return dao.getVo(id);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int saveVo(UserShopVo vo) {
        UserEntity userEntity = userDao.findByCode(vo.getUser().getCode());
        ShopEntity shopEntity = shopDao.findByCode(vo.getShop().getCode());
        if (userEntity == null || shopEntity == null) {
            throw new RRException("编号数据不匹配");
        }
        vo.getEntity().setUserId(userEntity.getId());
        vo.getEntity().setShopId(shopEntity.getId());
        return dao.save(vo.getEntity());
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int updateVo(UserShopVo vo) {
        return dao.update(vo.getEntity());
    }


    @Override
    public void downLoad(HttpServletResponse response, List<UserShopVo> list) {
        try {
            List<UserShopDown> datas = list.stream().map(item -> new UserShopDown(item)).collect(Collectors.toList());
            response.setContentType("application/vnd.ms-excel");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("测试", "UTF-8");
            response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
            EasyExcel.write(response.getOutputStream(), UserShopDown.class).sheet("模板").doWrite(datas);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    @Autowired
    private UserDao userDao;

    @Autowired
    private ShopDao shopDao;

    @Override
    public void upload(InputStream file) {
        EasyExcel.read(file, UserShopUp.class, new UploadDataListener()).sheet().doRead();
    }

    class UploadDataListener extends AnalysisEventListener<UserShopUp> {


        @Override
        @Transactional(rollbackFor = Exception.class)
        public void invoke(UserShopUp upVo, AnalysisContext analysisContext) {
            saveVo(upVo.toVo());
        }

        @Override
        public void doAfterAllAnalysed(AnalysisContext analysisContext) {

        }

    }


    /**内部方法**/


    /**外部方法实现**/

    @Override
    public List<UserEntity> findUserByShopId(Long shopId) {
        return dao.findUserByShopId(shopId);
    }
}
